#pragma once

#include <string>
#include <iostream>
#include <vector>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <atomic> //c++11
#include <sys/time.h>
#include <fstream>
#include <boost/algorithm/string.hpp>

namespace ns_util
{

    const std::string tem_path = "./temp/";

    class PathUtil
    {
    private:
        static std::string AddSuffix(const std::string &file_name, const std::string suffix)
        {
            std::string path = tem_path;
            path += file_name;
            path += suffix;
            return path;
        }

    public:
        // 1. 编译时使用
        // 为某个源文件路径添加上后缀: 1234 -> ./temp/1234.cpp
        static std::string Src(const std::string &file_name)
        {
            return AddSuffix(file_name, ".cpp");
        }
        // 为某个可执行程序加后缀: 1234 -> 1234.exe
        static std::string Exe(const std::string &file_name)
        {
            return AddSuffix(file_name, ".exe");
        }
        // 编译时报错信息
        static std::string CompilerError(const std::string &file_name)
        {
            return AddSuffix(file_name, ".compile_error");
        }

        // 2. 运行时使用
        static std::string Stdin(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdin");
        }
        static std::string Stdout(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdout");
        }
        static std::string Stderr(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stderr");
        }
    };
    class TimeUtil
    {
    public:
        // 获取当前时间的一个时间戳(只要秒级的即可)
        static std::string GetTimeStamp()
        {
            // 系统调用：int gettimeofday(struct timeval *tv, struct timezone *tz);
            // 第一个参数是输出型参数
            // 第二个是时区
            //    struct timeval
            //    {
            //          time_t tv_sec;       /* seconds */
            //          suseconds_t tv_usec; /* microseconds */
            //    };
            // time_t 就是时间戳类型（一个长整型）

            struct timeval _time;
            gettimeofday(&_time, nullptr);

            return std::to_string(_time.tv_sec);
        }

        // 获取毫秒级的时间戳
        static std::string GetTimeMs()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);

            return std::to_string(_time.tv_usec / 1000 + _time.tv_sec * 1000);
        }
    };

    class FileUtil
    {
    public:
        // 判断某个文件是否存在
        static bool IsFileExists(const std::string &file_name)
        {
            // 可以使用以 r 方式打开文件，打开失败就是不存在
            // 也可以使用一个系统调用 stat

            struct stat st;
            if (stat(file_name.c_str(), &st) == 0)
            {
                // 返回值：获取文件属性成功返回 0(即文件存在)， 否则返回 -1
                return true;
            }
            return false;
        }

        // 获得唯一的一个文件名
        static std::string UniqueFileName()
        {
            // 毫秒级时间戳 + 原子性递增唯一值 => 确保唯一性

            static std::atomic_uint id(0);
            id++;
            std::string uniq_id = std::to_string(id);

            std::string ms = TimeUtil::GetTimeMs();

            return ms + "_" + uniq_id;
        }

        static bool WriteFile(const std::string &filename, const std::string &content)
        {
            std::ofstream out(filename);
            if (!out.is_open())
            {
                return false;
            }

            // out << content;
            out.write(content.c_str(), content.size());

            out.close();
            return true;
        }
        // keep 参数为在读取文件时是否保留 \n (false：不保留)
        static bool ReadFile(const std::string &filename, std::string *content, bool keep = false)
        {
            content->clear();

            std::ifstream in(filename);
            if (!in.is_open())
            {
                return false;
            }
            std::string line;
            while (std::getline(in, line))
            {
                // getline 不保存 \n
                // 上面的判断是因为 ifstream 中有 operator bool
                *content += line;
                if (keep)
                    *content += '\n';
            }

            in.close();
            return true;
        }
    };

    class StringUtil
    {
    public:
        /*****************
         * str：要切割的字符串
         * target：切割后的结果
         * sep：分割符
         *****************/
        static void SplitString(const std::string &str, std::vector<std::string> *target, const std::string &sep)
        {
            // is_any_of:只要碰到字符在 sep 里面就切割
            // 最后一个参数：切割出的空的要不要压缩（on：压缩，off：不压缩）
            boost::split(*target, str, boost::is_any_of(sep), boost::algorithm::token_compress_on);
        }
    };

}